package com.mlh.binarytree.BFS;

import com.mlh.binarytree.TreeNode;

import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;

/**
 * @author 缪林辉
 * @date 2024/3/26 10:08
 * @DESCRIPTION
 */
// 给定一个二叉树的 根节点 root，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。
// 层序遍历之后，每层的最右节点。就是我们站在右侧能看到的所有节点
public class 二叉树的右视图 {
    public List<Integer> method1(TreeNode root) {
        if(root==null){return new ArrayList<>();}
        Deque<TreeNode> treeNodes = new LinkedList<>();
        List<Integer>res=new ArrayList<>();
        treeNodes.add(root);
        while(!treeNodes.isEmpty()){
            int len=treeNodes.size();
            while (len>0){
                TreeNode node=treeNodes.poll();
                len--;
                //遍历每层的时候，到最后len=0时，就是此层的最右节点
                if(len==0){
                    res.add(node.val);
                }
                if(node.left!=null){
                    treeNodes.add(node.left);
                }
                if(node.right!=null){
                    treeNodes.add(node.right);
                }
            }
        }
        return res ;
    }

    public List<Integer> practice(TreeNode root) {
        if(root==null){
            return new ArrayList<>();
        }
        LinkedList<TreeNode> queue=new LinkedList<>();
        queue.addLast(root);
        List<Integer>res=new ArrayList<>();
        while (!queue.isEmpty()){
            int queueCount=queue.size();
            res.add(queue.peekLast().val);
            while (queueCount>0){
                TreeNode node = queue.pollFirst();
                --queueCount;
                if(node.left!=null){
                    queue.addLast(node.left);
                }
                if(node.right!=null){
                    queue.addLast(node.right);
                }
            }
        }
        return res;
    }

    public List<Integer> practice2(TreeNode root) {
        if(root==null){
            return new ArrayList<>();
        }
        LinkedList<TreeNode>queue=new LinkedList<>();
        queue.addLast(root);
        List<Integer>res=new ArrayList<>();
        while(!queue.isEmpty()){
            int size=queue.size();
            res.add(queue.peekLast().val);
            for (int i = 0; i < size; i++) {
                TreeNode temp=queue.pollFirst();
                if(temp.left!=null){
                    queue.addLast(temp.left);
                }
                if(temp.right!=null){
                    queue.addLast(temp.right);
                }
            }
        }
        return res;
    }
}
